package org.lc.platform.base.vo;


import cn.hutool.core.collection.CollUtil;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.Data;

import java.util.Collections;
import java.util.List;
import java.util.function.Function;
import java.util.stream.Collectors;

@Data
@Schema(description = "分页结果")
public class PageVo<T> {
    @Schema(description = "总条数")
    private Long total;
    @Schema(description = "总页数")
    private Long pages;
    @Schema(description = "集合")
    private List<T> list;


    public static <PO, VO> PageVo<VO> of(Page<PO> p, Function<PO, VO> convertor) {
        PageVo<VO> dto = new PageVo<>();
        dto.setTotal(p.getTotal());
        dto.setPages(p.getPages());
        List<PO> records = p.getRecords();
        if (CollUtil.isEmpty(records)) {
            dto.setList(Collections.emptyList());
            return dto;
        }
        dto.setList(records.stream().map(convertor).collect(Collectors.toList()));
        return dto;

    }

    public static <VO> PageVo<VO> of(Page<VO> p) {
        PageVo<VO> dto = new PageVo<>();
        dto.setTotal(p.getTotal());
        dto.setPages(p.getPages());
        List<VO> records = p.getRecords();
        if (CollUtil.isEmpty(records)) {
            dto.setList(Collections.emptyList());
            return dto;
        }
        dto.setList(records);
        return dto;
    }
}
